"""
快速创建超级用户的脚本
使用方法：python create_superuser.py
"""
import os
import django

os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'travel_plan.settings')
django.setup()

from django.contrib.auth import get_user_model

User = get_user_model()

def create_superuser():
    username = input('请输入用户名: ')
    email = input('请输入邮箱（可选，直接回车跳过）: ') or ''
    password = input('请输入密码: ')
    password_confirm = input('请再次输入密码: ')
    
    if password != password_confirm:
        print('错误：两次输入的密码不一致！')
        return
    
    if User.objects.filter(username=username).exists():
        print(f'错误：用户名 "{username}" 已存在！')
        return
    
    try:
        user = User.objects.create_superuser(
            username=username,
            email=email,
            password=password
        )
        print(f'\n✅ 超级用户 "{username}" 创建成功！')
        print(f'现在可以使用此账号登录管理后台：http://localhost:8000/admin')
    except Exception as e:
        print(f'错误：创建用户失败 - {str(e)}')

if __name__ == '__main__':
    create_superuser()

